# -*- coding: utf-8 -*-
from django.template import Library
from bd42.apps.bdig.models import *
from django.db.models import Q

register = Library()

@register.inclusion_tag('adm/autor/trabalhos_pendentes.html')
def trabalhos_pendentes_autor(user):
    object_list = Trabalho.objects.filter(Q(autor = user), Q(workflow = 'cor_col_aut') | Q(workflow = 'cor_pub_aut') )
    return {'object_list' : object_list}

@register.inclusion_tag('adm/colaborador/trabalhos_pendentes.html')
def trabalhos_pendentes_colaborador(user):
    object_list = Trabalho.objects.filter(colaborador = user, workflow='cor_aut_col')
    return {'object_list' : object_list}

@register.inclusion_tag('adm/publicador/trabalhos_pendentes.html')
def trabalhos_pendentes_publicador(user):
    object_list = Trabalho.objects.filter(Q(workflow='cor_col_pub') | Q(workflow='cor_aut_pub'))
    return {'object_list' : object_list}
